package com.multisource.myself.config;

import com.multisource.myself.common.DataSourceType;

public class DataSourceContextHolder {
    // 存放当前线程使用的数据源类型
    private static final ThreadLocal<DataSourceType> DATA_SOURCE_HOLDER = new ThreadLocal<>();

    // 获取数据源
    public static DataSourceType getDataSource() {
        return DATA_SOURCE_HOLDER.get();
    }

    // 设置数据源
    public static void setDataSource(DataSourceType type) {
        DATA_SOURCE_HOLDER.set(type);
    }

    // 清除数据源
    public static void clearDataSource() {
        DATA_SOURCE_HOLDER.remove();
    }
}
